package com.deguru.main.utils;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

import javax.imageio.ImageIO;

public class GetVerifyCode {

	private int wide = 100;
	private int height = 50;
	private String text;
	private Random rdm = new Random();
	private String[] fontNames = { "宋体", "华文楷体", "黑体", "微软雅黑", "楷体_GB2312" }; // 字体数组
	private String codes = "23456789abcdefghjkmnpqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ"; // 验证码数组

	private Color randomColor(int i) {// 获取随机的颜色
		int r = rdm.nextInt(50)+i; // 这里为什么是150，因为当r，g，b都为255时，即为白色，为了好辨认，需要颜色深一点。
		int g = rdm.nextInt(50)+i;
		int b = rdm.nextInt(50)+i;
		return new Color(r, g, b); // 返回一个随机颜色
	}

	private Font randomFont() {// 获取随机字体
		int index = rdm.nextInt(fontNames.length); // 获取随机的字体
		String fontName = fontNames[index];
		int style = rdm.nextInt(4); // 随机获取字体的样式，0是无样式，1是加粗，2是斜体，3是加粗加斜体
		int size = rdm.nextInt(5) + 24; // 随机获取字体的大小
		return new Font(fontName, style, size); // 返回一个随机的字体
	}

	private char randomChar() {// 获取随机字符
		int index = rdm.nextInt(codes.length());
		return codes.charAt(index);
	}

	private void drawLine(BufferedImage image) {// 画干扰线，验证码干扰线用来防止计算机解析图片
		int num = 3; // 定义干扰线的数量
		Graphics2D g = (Graphics2D) image.getGraphics();
		for (int i = 0; i < num; i++) {
			int x1 = rdm.nextInt(wide);
			int y1 = rdm.nextInt(height);
			int x2 = rdm.nextInt(wide);
			int y2 = rdm.nextInt(height);
			g.setColor(randomColor(100));
			g.drawLine(x1, y1, x2, y2);
		}
	}

	private BufferedImage createImage() {// 创建图片的方法
		BufferedImage image = new BufferedImage(wide, height, BufferedImage.TYPE_INT_RGB); // 创建图片缓冲区
		Graphics2D g = (Graphics2D) image.getGraphics(); // 获取画笔
		g.setColor(randomColor(200)); // 设置背景色
		g.fillRect(0, 0, wide, height);
		return image; // 返回一个图片
	}

	public BufferedImage getImage() {// 获取验证码图片的方法

		BufferedImage image = createImage();
		Graphics2D g = (Graphics2D) image.getGraphics(); // 获取画笔
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < 4; i++) // 画四个字符即可
		{
			String s = randomChar() + ""; // 随机生成字符，因为只有画字符串的方法，没有画字符的方法，所以需要将字符变成字符串再画
			sb.append(s); // 添加到StringBuilder里面
			float x = i * 1.0F * wide / 4; // 定义字符的x坐标
			g.setFont(randomFont()); // 设置字体，随机
			g.setColor(randomColor(100)); // 设置颜色，随机
			g.drawString(s, x, height - 5);
		}
		this.text = sb.toString();
		drawLine(image);
		return image;
	}

	public String getText() {// 获取验证码文本的方法
		return text;
	}

	public static void output(BufferedImage image, OutputStream out) throws IOException {// 将验证码图片写出的方法
		ImageIO.write(image, "JPEG", out);
	}
}
